package com.liuguangzhou.project.controller;

import org.springframework.stereotype.Component;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.util.concurrent.CopyOnWriteArraySet;

/**
 * @ClassName WebSocketServerTwo
 * @Description: TODO
 * @Author liuguangzhou
 * @Date 2020/3/2
 * @Version V1.0
 **/

/*@ServerEndpoint注解是一个类层次的注解，它的功能主要是将目前的类定义成一个websocket服务器端,
 * 注解的值将被用于监听用户连接的终端访问URL地址,客户端可以通过这个URL来连接到WebSocket服务器端
 */


//@ServerEndpoint("/websocket/{userID}")，括号中的内容就是客户端请求Socket连接时的访问路径，userID是我要求客户端传来的参数，我这里算是为了标识该客户端吧。
//在该类中添加属性 userID，并添加对应的getUserID()方法。
//在onOpen()方法即建立连接的时候就接收参数userID，需要标识@PathParam("userID") 。接收参数后直接赋值给属性userID。
//是针对特定客户端发送消息。服务器和客户端在建立连接成功后就生成了一个WebSocket对象，并存在集合中，对象里特有的属性是我们设置的userID。所以通过唯一的userID就能标识服务器与该客户端建立的那个连接啦！这样要求发送消息时，传入userID与消息，服务器在自己的WebSocket连接集合中遍历找到对应客户端的连接，就可以直接发消息过去啦~~

@ServerEndpoint("/websocket/{userID}")
@Component
public class WebSocketControllerTwo {
    //每个客户端都会有相应的session,服务端可以发送相关消息
    private Session session;

    //接收userID
    private Integer userID;

    //J.U.C包下线程安全的类，主要用来存放每个客户端对应的webSocket连接
    private static CopyOnWriteArraySet<WebSocketControllerTwo> copyOnWriteArraySet = new CopyOnWriteArraySet<WebSocketControllerTwo>();

    public Integer getUserID() {
        return userID;
    }

    /**
    * 功能描述:
    * 〈打开连接。进入页面后会自动发请求到此进行连接〉
    * @Param: [session, userID]
    * @Return: void
    * @Author: liuguangzhou
    * @Date: 2020/3/2 21:29
    */
    @OnOpen
    public void onOpen(Session session, @PathParam("userID") Integer userID) {
        this.session = session;
        this.userID = userID;
        System.out.println(this.session.getId());
        copyOnWriteArraySet.add(this);
    }

    /**
    * 功能描述:
    * 〈用户关闭页面，即关闭连接〉
    * @Param: []
    * @Return: void
    * @Author: liuguangzhou
    * @Date: 2020/3/2 21:28
    */
    @OnClose
    public void onClose() {
        copyOnWriteArraySet.remove(this);
    }


    /**
    * 功能描述:
    * 〈测试客户端发送消息，测试是否联通〉
    * @Param: [message]
    * @Return: void
    * @Author: liuguangzhou
    * @Date: 2020/3/2 21:28
    */
    @OnMessage
    public void onMessage(String message) {
        System.out.println("websocket收到客户端发来的消息:"+message);
    }

   /**
   * 功能描述:
   * 〈出现错误〉
   * @Param: [session, error]
   * @Return: void
   * @Author: liuguangzhou
   * @Date: 2020/3/2 21:28
   */
    @OnError
    public void onError(Session session, Throwable error) {
        System.out.println("发生错误：" + error.getMessage() + "; sessionId:" + session.getId());
        error.printStackTrace();
    }

    public void sendMessage(Object object){
        //遍历客户端
        for (WebSocketControllerTwo webSocket : copyOnWriteArraySet) {
            System.out.println("websocket广播消息：" + object.toString());
            try {
                //服务器主动推送
                webSocket.session.getBasicRemote().sendObject(object) ;
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    /**
    * 功能描述:
    * 〈〉
    * @Param: [message]
    * @Return: void
    * @Author: liuguangzhou
    * @Date: 2020/3/2 21:27
    */
    public void sendMessage(String message) {
        //遍历客户端
        for (WebSocketControllerTwo webSocket : copyOnWriteArraySet) {
            System.out.println("websocket广播消息：" + message);
            try {
                //服务器主动推送
                webSocket.session.getBasicRemote().sendText(message);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    /**
    * 功能描述:
    * 〈发送消息 根据ID〉
    * @Param: [userID, message]
    * @Return: void
    * @Author: liuguangzhou
    * @Date: 2020/3/2 21:27
    */
    public void sendMessage(Integer userID, String message) throws Exception {
        Session session = null;
        WebSocketControllerTwo tempWebSocket = null;
        for (WebSocketControllerTwo webSocket : copyOnWriteArraySet) {
            if (webSocket.getUserID() == userID) {
                tempWebSocket = webSocket;
                session = webSocket.session;
                break;
            }
        }
        if (session != null) {
            //服务器主动推送
            tempWebSocket.session.getBasicRemote().sendText(message);
        } else {
            System.out.println("没有找到你指定ID的会话：{}"+ "; userId:" + userID);
        }
    }
}
